package com.ruoyi.ade.model;

import com.ruoyi.ade.llm.chunk.IAbstractChunkRequest;
import com.ruoyi.ade.llm.model.EmbeddingRequest;

/**
 * 从 ADE检索库中检索 请求对象
 */
public class AdeSearchRequest {

    //文本分割方法   默认按照标点符号分割
    private String chunkType = IAbstractChunkRequest.SPLIT_MARK;
    private String embedModel = EmbeddingRequest.MODEL_DMETS ;
    //知识库ID
    private Long adekb;
    //相似度阈值
    private float simScore = 0.5f;
    //topN 检索结果数量
    private int topN = 5;
    //请求的文本
    private String emr;

    private boolean debug;

    public String getChunkType() {
        return chunkType;
    }

    public void setChunkType(String chunkType) {
        this.chunkType = chunkType;
    }

    public String getEmbedModel() {
        return embedModel;
    }

    public void setEmbedModel(String embedModel) {
        this.embedModel = embedModel;
    }

    public Long getAdekb() {
        return adekb;
    }

    public void setAdekb(Long adekb) {
        this.adekb = adekb;
    }

    public float getSimScore() {
        return simScore;
    }

    public void setSimScore(float simScore) {
        this.simScore = simScore;
    }

    public int getTopN() {
        return topN;
    }

    public void setTopN(int topN) {
        this.topN = topN;
    }

    public String getEmr() {
        return emr;
    }

    public void setEmr(String emr) {
        this.emr = emr;
    }

    public boolean isDebug() {
        return debug;
    }

    public void setDebug(boolean debug) {
        this.debug = debug;
    }
}
